NUM_GPUS=2

python3 -m torch.distributed.launch --nproc_per_node $NUM_GPUS --master_port 13325 --use_env \
    train.py \
    --me tlggnv1 \
    --dataset_file ytvosv1 \
    --with_box_refine \
    --binary \
    --batch_size 1 \
    --epochs 6 \
    --lr_drop 3 5 \
    --num_frames 8 \
    --backbone video_swin_t_p4w7 \
    --backbone_pretrained ./weights/swin_tiny_patch244_window877_kinetics400_1k.pth \
    --output_dir ./results/tlggnv1Q10K10 \
    --K 10 \
    --num_queries 10

python inference_ytvos1.py \
    --me tlggnv1 \
    --dataset_file ytvos \
    --backbone video_swin_t_p4w7 \
    --backbone_pretrained ./weights/swin_tiny_patch244_window877_kinetics400_1k.pth \
    --output_dir ./results/tlggnv1Q10K10 \
    --resume ./results/tlggnv1Q10K10/checkpoint.pth \
    --K 10 \
    --num_queries 10 